/*
 *  Copyright 2019-2020 Zheng Jie
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package cn.com.css.taxInspection.util;

import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.poi.excel.BigExcelWriter;
import cn.hutool.poi.excel.ExcelUtil;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.util.IOUtils;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.security.MessageDigest;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * File工具类，扩展 hutool 工具包
 */
public class FileUtil extends cn.hutool.core.io.FileUtil {

	private static final Logger log = LoggerFactory.getLogger(FileUtil.class);

	/**
	 * 系统临时目录 <br>
	 * windows 包含路径分割符，但Linux 不包含, 在windows \\==\ 前提下， 为安全起见 同意拼装 路径分割符，
	 * 
	 * <pre>
	 *       java.io.tmpdir
	 *       windows : C:\Users/xxx\AppData\Local\Temp\
	 *       linux: /temp
	 * </pre>
	 */
	public static final String SYS_TEM_DIR = System.getProperty("java.io.tmpdir") + File.separator;
	/**
	 * 定义GB的计算常量
	 */
	private static final int GB = 1024 * 1024 * 1024;
	/**
	 * 定义MB的计算常量
	 */
	private static final int MB = 1024 * 1024;
	/**
	 * 定义KB的计算常量
	 */
	private static final int KB = 1024;

	/**
	 * 格式化小数
	 */
	private static final DecimalFormat DF = new DecimalFormat("0.00");

	public static final String IMAGE = "图片";
	public static final String TXT = "文档";
	public static final String MUSIC = "音乐";
	public static final String VIDEO = "视频";
	public static final String OTHER = "其他";

	/**
	 * MultipartFile转File
	 */
	public static File toFile(MultipartFile multipartFile) {
		// 获取文件名
		String fileName = multipartFile.getOriginalFilename();
		// 获取文件后缀
		String prefix = "." + getExtensionName(fileName);
		File file = null;
		try {
			// 用uuid作为文件名，防止生成的临时文件重复
			file = new File(SYS_TEM_DIR + IdUtil.simpleUUID() + prefix);
			// MultipartFile to File
			multipartFile.transferTo(file);
		} catch (IOException e) {
			log.error(e.getMessage(), e);
		}
		return file;
	}

	/**
	 * 获取文件扩展名，不带 .
	 */
	public static String getExtensionName(String filename) {
		if ((filename != null) && (filename.length() > 0)) {
			int dot = filename.lastIndexOf('.');
			if ((dot > -1) && (dot < (filename.length() - 1))) {
				return filename.substring(dot + 1);
			}
		}
		return filename;
	}

	/**
	 * Java文件操作 获取不带扩展名的文件名
	 */
	public static String getFileNameNoEx(String filename) {
		if ((filename != null) && (filename.length() > 0)) {
			int dot = filename.lastIndexOf('.');
			if ((dot > -1) && (dot < (filename.length()))) {
				return filename.substring(0, dot);
			}
		}
		return filename;
	}

	/**
	 * 文件大小转换
	 */
	public static String getSize(long size) {
		String resultSize;
		if (size / GB >= 1) {
			// 如果当前Byte的值大于等于1GB
			resultSize = DF.format(size / (float) GB) + "GB   ";
		} else if (size / MB >= 1) {
			// 如果当前Byte的值大于等于1MB
			resultSize = DF.format(size / (float) MB) + "MB   ";
		} else if (size / KB >= 1) {
			// 如果当前Byte的值大于等于1KB
			resultSize = DF.format(size / (float) KB) + "KB   ";
		} else {
			resultSize = size + "B   ";
		}
		return resultSize;
	}

	/**
	 * inputStream 转 File
	 */
	static File inputStreamToFile(InputStream ins, String name) throws Exception {
		File file = new File(SYS_TEM_DIR + name);
		if (file.exists()) {
			return file;
		}
		OutputStream os = new FileOutputStream(file);
		int bytesRead;
		int len = 8192;
		byte[] buffer = new byte[len];
		while ((bytesRead = ins.read(buffer, 0, len)) != -1) {
			os.write(buffer, 0, bytesRead);
		}
		os.close();
		ins.close();
		return file;
	}

	/**
	 * 将文件名解析成文件的上传路径
	 */
	public static File upload(MultipartFile file, String filePath) {
		Date date = new Date();
		SimpleDateFormat format = new SimpleDateFormat("yyyyMMddhhmmssS");
		String name = getFileNameNoEx(file.getOriginalFilename());
		String suffix = getExtensionName(file.getOriginalFilename());
		String nowStr = "-" + format.format(date);
		try {
			String fileName = name + nowStr + "." + suffix;
			String path = filePath + fileName;
			// getCanonicalFile 可解析正确各种路径
			File dest = new File(path).getCanonicalFile();
			// 检测是否存在目录
			if (!dest.getParentFile().exists()) {
				if (!dest.getParentFile().mkdirs()) {
					System.out.println("was not successful.");
				}
			}
			// 文件写入
			file.transferTo(dest);
			return dest;
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		}
		return null;
	}

	/**
	 * 导出excel
	 */
	public static void downloadExcel(List<Map<String, Object>> list, HttpServletResponse response) throws IOException {
		String tempPath = SYS_TEM_DIR + IdUtil.fastSimpleUUID() + ".xlsx";
		File file = new File(tempPath);
		BigExcelWriter writer = ExcelUtil.getBigWriter(file);
		// 一次性写出内容，使用默认样式，强制输出标题
		writer.write(list, true);
		SXSSFSheet sheet = (SXSSFSheet) writer.getSheet();
		// 上面需要强转SXSSFSheet 不然没有trackAllColumnsForAutoSizing方法
		sheet.trackAllColumnsForAutoSizing();
		// 列宽自适应
		writer.autoSizeColumnAll();
		// response为HttpServletResponse对象
		response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
		// test.xls是弹出下载对话框的文件名，不能为中文，中文请自行编码
		response.setHeader("Content-Disposition", "attachment;filename=file.xlsx");
		ServletOutputStream out = response.getOutputStream();
		// 终止后删除临时文件
		file.deleteOnExit();
		writer.flush(out, true);
		// 此处记得关闭输出Servlet流
		IoUtil.close(out);
	}

	public static String getFileType(String type) {
		String documents = "txt doc pdf ppt pps xlsx xls docx";
		String music = "mp3 wav wma mpa ram ra aac aif m4a";
		String video = "avi mpg mpe mpeg asf wmv mov qt rm mp4 flv m4v webm ogv ogg";
		String image = "bmp dib pcp dif wmf gif jpg tif eps psd cdr iff tga pcd mpt png jpeg";
		if (image.contains(type)) {
			return IMAGE;
		} else if (documents.contains(type)) {
			return TXT;
		} else if (music.contains(type)) {
			return MUSIC;
		} else if (video.contains(type)) {
			return VIDEO;
		} else {
			return OTHER;
		}
	}

	public static void checkSize(long maxSize, long size) throws Exception {
		// 1M
		int len = 1024 * 1024;
		if (size > (maxSize * len)) {
			throw new Exception("文件超出规定大小");
		}
	}

	/**
	 * 判断两个文件是否相同
	 */
	public static boolean check(File file1, File file2) {
		String img1Md5 = getMd5(file1);
		String img2Md5 = getMd5(file2);
		return img1Md5.equals(img2Md5);
	}

	/**
	 * 判断两个文件是否相同
	 */
	public static boolean check(String file1Md5, String file2Md5) {
		return file1Md5.equals(file2Md5);
	}

	@SuppressWarnings("resource")
	private static byte[] getByte(File file) {
		// 得到文件长度
		byte[] b = new byte[(int) file.length()];
		try {
			InputStream in = new FileInputStream(file);
			try {
				System.out.println(in.read(b));
			} catch (IOException e) {
				log.error(e.getMessage(), e);
			}
		} catch (FileNotFoundException e) {
			log.error(e.getMessage(), e);
			return null;
		}
		return b;
	}

	private static String getMd5(byte[] bytes) {
		// 16进制字符
		char[] hexDigits = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };
		try {
			MessageDigest mdTemp = MessageDigest.getInstance("MD5");
			mdTemp.update(bytes);
			byte[] md = mdTemp.digest();
			int j = md.length;
			char[] str = new char[j * 2];
			int k = 0;
			// 移位 输出字符串
			for (byte byte0 : md) {
				str[k++] = hexDigits[byte0 >>> 4 & 0xf];
				str[k++] = hexDigits[byte0 & 0xf];
			}
			return new String(str);
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		}
		return null;
	}

	/**
	 * 下载文件
	 *
	 * @param request
	 *            /
	 * @param response
	 *            /
	 * @param file
	 *            /
	 */
	public static void downloadFile(HttpServletRequest request, HttpServletResponse response, File file,
			boolean deleteOnExit) {
		response.setCharacterEncoding(request.getCharacterEncoding());
		response.setContentType("application/octet-stream");
		FileInputStream fis = null;
		try {
			fis = new FileInputStream(file);
			response.setHeader("Content-Disposition", "attachment; filename=" + file.getName());
			IOUtils.copy(fis, response.getOutputStream());
			response.flushBuffer();
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		} finally {
			if (fis != null) {
				try {
					fis.close();
					if (deleteOnExit) {
						file.deleteOnExit();
					}
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
		}
	}

	public static String getMd5(File file) {
		return getMd5(getByte(file));
	}

	//工具方法:获取表格值
	public static boolean isRowEmpty(Row row) {
		if (row == null) return true;
		for (int c = 0; c <= row.getLastCellNum(); c++) {
			Cell cell = row.getCell(c);
			if (cell != null && cell.getCellType() != CellType.BLANK) {
				if (cell.getCellType() == CellType.STRING && !cell.getStringCellValue().trim().isEmpty()) return false;
				if (cell.getCellType() == CellType.NUMERIC || cell.getCellType() == CellType.BOOLEAN) return false;
			}
		}
		return true;
	}

	// 工具方法：读取字符串单元格
	public static String getStringCellValue(Cell cell) {
		if (cell == null) return "";
		switch (cell.getCellType()) {
			case STRING:
				return cell.getStringCellValue().trim();
			case NUMERIC:
				// 判断是否为整数
				double numValue = cell.getNumericCellValue();
				if (numValue == (long) numValue) {
					return String.valueOf((long) numValue);  // 整数返回无小数点
				} else {
					return String.valueOf(numValue);  // 小数返回原样
				}
			case BOOLEAN:
				return String.valueOf(cell.getBooleanCellValue());
			case FORMULA:
				try {
					return cell.getStringCellValue();
				} catch (IllegalStateException e) {
					// 公式计算结果为数字时
					double formulaNumValue = cell.getNumericCellValue();
					if (formulaNumValue == (long) formulaNumValue) {
						return String.valueOf((long) formulaNumValue);
					} else {
						return String.valueOf(formulaNumValue);
					}
				}
			case BLANK:
			case ERROR:
			default:
				return "";
		}
	}

	// 工具方法：读取数字单元格
	public static Double getDoubleCellValue(Cell cell) {
		if (cell == null) return null;
		try {
			if (cell.getCellType() == CellType.NUMERIC) {
				return cell.getNumericCellValue();
			} else if (cell.getCellType() == CellType.STRING) {
				String val = cell.getStringCellValue().trim();
				if (!val.isEmpty()) {
					return Double.parseDouble(val);
				}
			}
		} catch (Exception e) {
			log.debug("getDoubleCellValue error:{}", e.getMessage());
		}
		return null;
	}

	// 工具方法：读取日期单元格
	public static Date getDateCellValue(Cell cell) {
		if (cell == null) return null;
		try {
			if (cell.getCellType() == CellType.NUMERIC && DateUtil.isCellDateFormatted(cell)) {
				return cell.getDateCellValue();
			} else if (cell.getCellType() == CellType.STRING) {
				String dateStr = cell.getStringCellValue().trim();
				if (dateStr.isEmpty()) return null;

				// 支持多种常见格式
				List<String> patterns = Arrays.asList("yyyy-MM-dd HH:mm:ss", "yyyy-MM-dd", "yyyy/MM/dd HH:mm:ss", "yyyy/MM/dd");
				for (String pattern : patterns) {
					try {
						return new SimpleDateFormat(pattern).parse(dateStr);
					} catch (Exception ignored) {
					}
				}
			}
		} catch (Exception ex) {
			log.debug("getDateCellValue error:{}", ex.getMessage());
		}
		return null;
	}

}
